import pandas as pd

# 创建示例 DataFrame
data = {
    'date': ['2024-01-01', '2024-01-01', '2024-02-01', '2024-02-01'],  # 长度为 4
    'category': ['A', 'B', 'A', 'B'],  # 长度为 4
    'sales': [100, 200, 150, 250],  # 长度为 4
    'id': [1, 2, 3, 4],  # 长度为 4
    'col1': [10, 20, 30, 40],  # 长度为 4
    'col2': [40, 50, 60, 70]   # 长度为 4
}

df = pd.DataFrame(data)

# 1. 数据透视表
pivot_df = df.pivot(index='date', columns='category', values='sales')
print(pivot_df)
print("===========数据透视表示例代码===========")

# 2. 数据透视表聚合
pivot_table_df = df.pivot_table(index='date', columns='category', values='sales', aggfunc='sum')
print(pivot_table_df)
print("===========数据透视表聚合示例代码===========")

# 3. 重塑数据框（长宽转换）
melted_df = df.melt(id_vars='id', value_vars=['col1', 'col2'], var_name='variable', value_name='value')
print(melted_df)
print("===========重塑数据框（长宽转换）示例代码===========")